import argparse


def get_parser():
    parser = argparse.ArgumentParser(description="segmentation test")
    parser.add_argument('--project_name', type=str, default="图像分割")
    parser.add_argument('--use_cuda', type=bool, default=True)
    parser.add_argument('--seed', type=int, default=123)
    parser.add_argument('--data_base', type=str, default='E:/Datasets2/lane_seg')
    parser.add_argument('--resume', type=bool, default=True)
    parser.add_argument('--pretrained_model', type=str, default='./weights/')
    parser.add_argument('--lr',type=float,default=0.0001)
    parser.add_argument('--milestones', type=list, default=[50, 80])
    parser.add_argument('--epoches', type=int, default=200)
    parser.add_argument('--save_path',type=str,default="./weights/")

    args = parser.parse_args()
    return args
